import torch
a=torch.cuda.device_count()
print(a)
b=torch.cuda.is_available()
print(b)
print(torch.__version__)